package models

import (
	"fmt"
	"os"

	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

// 定义常量DB常量和err常量
var DB *gorm.DB
var DBErr error

// sqlite数据库连接
func init() {
	fmt.Println("数据库开始连接")
	DB, DBErr = gorm.Open(sqlite.Open("./models/db/my_db_01.s3db"), &gorm.Config{})
	// DB, DBErr = gorm.Open(mysql.Open("root:123456@tcp(127.0.0.1:3306)/my_db_02?charset=utf8mb4&parseTime=True&loc=Local"), &gorm.Config{})
	if DBErr != nil {
		fmt.Println("加载数据库失败", DBErr)
		os.Exit(1)
	}
	// 初始化创建表
	// DB.AutoMigrate(&Menu{}, &Role{}, &User{}, &Department{})
	// DB.AutoMigrate(&Department{})
	// 插入模拟数据
	// InsertMockRoles()
	// InsertMockUser()
	// InsertMockURs()
	// InsertMockMenus()
	// InsertMockRMs()
	// InsertMockDepartment()
}
